"""接口自动化"""
"""
学习目标:
概念:
语法(操作):
使用jsonpath
1.先导入jsonpath
from jsonpath import jsonpath

2.语法:
jsonpath(source,"$..key")
source:表示字典名

"$..乌鲁木齐"表示如下:
(1).$..  :固定格式
(2).乌鲁木齐 : 是指字典的键名称
"""
import jsonpath
class GetKeyWord():
    @staticmethod
    def getKeyWord(source_data,keyword):
        try:
            return jsonpath.jsonpath(source_data,f"$..{keyword}")[0]
        except :
            print(f'关键字{keyword}不存在')
            return False

    @staticmethod
    def getKeyWords(source_data, keyword):
        try:
            return jsonpath.jsonpath(source_data, f"$..{keyword}")
        except:
            print(f'关键字{keyword}不存在')
            return False
if __name__ == '__main__':
    d1 = {'苹果': 5.98, '梨子': 4.56, '香蕉': 2.5, '橘子': 6.3,
          '葡萄1': {'新疆': {"吐鲁番": 999, "乌鲁木齐": 991}},
          '葡萄2': {'新疆': {"吐鲁番": 998, "乌鲁木齐": 99}} ,'四川': 201}

    print(GetKeyWord.getKeyWord(d1, "乌鲁木齐"))
    print(GetKeyWord.getKeyWords(d1, "乌鲁木齐"))
